<template>
  <div>
    <vxe-grid v-bind="gridOptions">
      <template #edit_name="{ row }">
        <vxe-input v-model="row.name"></vxe-input>
      </template>
      <template #default_name="{ row }">
        <span>{{ row.name }}</span>
      </template>

      <template #region_edit="{ row }">
        <vxe-tree-select v-model="row.region" :options="regionOptions"></vxe-tree-select>
      </template>
      <template #region_default="{ row }">
        <span>{{ formatRegionLabel(row.region) }}</span>
      </template>
      <template #regionList_edit="{ row }">
        <vxe-tree-select v-model="row.regionList" :options="regionListOptions" multiple></vxe-tree-select>
      </template>
      <template #regionList_default="{ row }">
        <span>{{ formatRegionListLabel(row.regionList) }}</span>
      </template>
    </vxe-grid>
  </div>
</template>

<script lang="ts">
import Vue from 'vue'
import type { VxeGridProps } from 'vxe-table'
import type { VxeTreeSelectPropTypes } from 'vxe-pc-ui'
import XEUtils from 'xe-utils'

interface RowVO {
  id: number
  name: string
  role: string
  region: string
  regionList: string[]
}

export default Vue.extend({
  data () {
    const gridOptions: VxeGridProps<RowVO> = {
      border: true,
      showOverflow: true,
      editConfig: {
        trigger: 'click',
        mode: 'row'
      },
      columns: [
        { type: 'seq', width: 70 },
        { field: 'name', title: 'Name', minWidth: 200, editRender: { autoFocus: 'input' }, slots: { edit: 'edit_name', default: 'default_name' } },
        { field: 'region', title: '下拉树单选', width: 200, editRender: {}, slots: { edit: 'region_edit', default: 'region_default' } },
        { field: 'regionList', title: '下拉树多选', width: 200, editRender: {}, slots: { edit: 'regionList_edit', default: 'regionList_default' } }
      ],
      data: [
        { id: 10001, name: 'Test1', role: 'Develop', region: '', regionList: [] },
        { id: 10002, name: 'Test2', role: 'Test', region: '2-1', regionList: ['1-2', '2-1'] },
        { id: 10003, name: 'Test3', role: 'PM', region: '', regionList: [] }
      ]
    }

    const regionOptions: VxeTreeSelectPropTypes.Options = [
      {
        label: '广东省',
        value: '1',
        children: [
          { label: '深圳市', value: '1-1' },
          { label: '广州市', value: '1-2' }
        ]
      },
      {
        label: '北京',
        value: '2',
        children: [
          { label: '东城区', value: '2-1' },
          { label: '西城区', value: '2-2' }
        ]
      }
    ]

    const regionListOptions: VxeTreeSelectPropTypes.Options = [
      {
        label: '广东省',
        value: '1',
        children: [
          { label: '深圳市', value: '1-1' },
          { label: '广州市', value: '1-2' }
        ]
      },
      {
        label: '北京',
        value: '2',
        children: [
          { label: '东城区', value: '2-1' },
          { label: '西城区', value: '2-2' }
        ]
      }
    ]

    return {
      gridOptions,
      regionOptions,
      regionListOptions
    }
  },
  methods: {
    formatRegionLabel (val: string) {
      const rest = XEUtils.findTree(this.regionOptions, item => item.value === val, { children: 'children' })
      return rest ? rest.item.label : val
    },
    formatRegionListLabel (vals: string[]) {
      if (vals) {
        return vals.map(val => {
          const rest = XEUtils.findTree(this.regionListOptions, item => item.value === val, { children: 'children' })
          return rest ? rest.item.label : val
        }).join(',')
      }
      return ''
    }
  }
})
</script>
